Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Cuda Device Selection #311

Merged
merged 11 commits into from
Jun 13, 2024
Merged

Cuda Device Selection #311

merged 11 commits into from
Jun 13, 2024

Conversation

hym97
Copy link
Contributor

@hym97 hym97 commented May 29, 2024

Adding GPU selection features and one unit test, and updating the documentation.

Adding more test cases to make the feature more robust.

For CLI usage:

TotalSegmentator .... -d gpu:X (e.g. gpu:1 -> cuda:1)`

For Python API:

device = 'gpu:1' # cuda:1
totalsegmentator(input_path, output_path, device=device)

@hym97 hym97 closed this Jun 7, 2024
@hym97 hym97 reopened this Jun 7, 2024
@hym97
Copy link
Contributor Author

hym97 commented Jun 7, 2024

Fixed a bug for python_api call.

@wasserth
Copy link
Owner

wasserth commented Jun 7, 2024

Thanks for the PR. Currently I am a bit busy. I will look at it as soon as possible.

@wasserth wasserth merged commit 71d3584 into wasserth:master Jun 13, 2024
@hym97 hym97 deleted the cuda_device_selection branch June 17, 2024 21:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@hym97 @wasserth